import django_filters

from HttpTestcase.models import TestsuiteSet2Testsuite


class suite_set_suiteFilters(django_filters.rest_framework.FilterSet):
    testsuite_set = django_filters.CharFilter(method='testsuite_set_filter', lookup_expr='exact')

    def testsuite_set_filter(self, queryset, name, value):
        return queryset.filter(testsuite_set_id=value)

    class Mete:
        model = TestsuiteSet2Testsuite
        fields = ['testsuite_set']